
#include "display.h"
#include "regex.h"
#include "terrainTable.h"

TerrainTable::TerrainTable( void ) {
  add( new TerrainArena() );
  add( new TerrainBeach() );
  add( new TerrainCity() );
  add( new TerrainChurch() );
  add( new TerrainDeepwater() );
  add( new TerrainDesert() );
  add( new TerrainField() );
  add( new TerrainForest() );
  add( new TerrainGraveyard() );
  add( new TerrainIndoor() );
  add( new TerrainJungle() );
  add( new TerrainMine() );
  add( new TerrainMountain() );
  add( new TerrainPlains() );
  add( new TerrainRoad() );
  add( new TerrainStone() );
  add( new TerrainSwamp() );
  add( new TerrainTreetop() );
  add( new TerrainUnderwater() );
  add( new TerrainVillage() );
  add( new TerrainVolcano() );
  add( new TerrainWater() );
  return;
}

TerrainTable::~TerrainTable( void ) {
  return;
}

void TerrainTable::add( Terrain* t ) { // public
  terrain().insert( std::make_pair( t->name(), t ) );
  return;
}

Terrain* TerrainTable::find( const std::string& name ) const { // public
  std::map<std::string,Terrain*>::const_iterator it;
  if ( ( it = terrain().find( name ) ) != terrain().end() ) {
    return it->second;
  }
  for ( it = terrain().begin(); it != terrain().end(); ++it ) {
    if ( Regex::strPrefix( name, it->second->name() ) ) {
      return it->second;
    }
  }
  return terrain().begin()->second;
}

std::string TerrainTable::list( void ) const { // public
  std::vector<std::string> items;
  for ( std::map<std::string,Terrain*>::const_iterator it = terrain().begin(); it != terrain().end(); ++it ) {
    items.push_back( it->second->name() );
  }
  return Display::formatColumns( 3, items, 79 );
}

